package tetris.agent;

import tetris.simulator.State;
import tetris.features.FeatureExtraction;
import java.util.*;

public class RewardFunctions {
	
	public double rowsCleared(State s){
		State newState = new State(s);
		return (double)newState.getRowsCleared();
	}
	
	public double depthFromTop(State s){
		FeatureExtraction features = new FeatureExtraction();
		State newState = new State(s);
		
		double maxHeight = features.maxtop(newState);
		return (double)State.ROWS - maxHeight;		
	}

}
